from jmeter_api.assertions.response.elements import ResponseAssertion

MODULE_CLASS = {
    "TestPlan": "jmeter_api.non_test_elements.test_plan.elements",
    "ThreadGroupAction": "jmeter_api.basics.thread_group.elements",
    "HTTPCacheManager": "jmeter_api.configs.http_cache_manager.elements",
    "CommonThreadGroup": "jmeter_api.thread_groups.common_thread_group.elements",
    "ConcurrencyThreadGroup": "jmeter_api.thread_groups.concurrency_thread_group.elements",
    "HttpRequest": "jmeter_api.samplers.http_request.elements",
    "Method": "jmeter_api.samplers.http_request.elements",
    "Protocol": "jmeter_api.samplers.http_request.elements",
    "Implement": "jmeter_api.samplers.http_request.elements",
    "Source": "jmeter_api.samplers.http_request.elements",
    "ResponseAssertion": "jmeter_api.assertions.response.elements",
    "TestField": "jmeter_api.assertions.response.elements",
    "TestType": "jmeter_api.assertions.response.elements",
    "Scope": "jmeter_api.assertions.response.elements",
    "UserDefineVariables": "jmeter_api.configs.user_defined_variables.elements",
    "Argument": "jmeter_api.configs.user_defined_variables.elements",
    "ViewResultsTree": "jmeter_api.listeners.view_results_tree.elements",
    "ConstantTimer": "jmeter_api.timers.constant_timer.elements",
    "JSR223PreProcessor": "jmeter_api.pre_processors.jsr223.elements",
    "ScriptLanguage": "jmeter_api.basics.jsr223",
}